package cn.minimelon.api.utils.captcha;

import java.awt.*;
import java.awt.image.BufferedImage;
import java.util.Random;

/**
 * 图片验证码生成
 */
public class ImageVerificationCode {
    private static final int CAPTCHA_IMAGE_WIDTH = 160; //验证码图片的宽
    private static final int CAPTCHA_IMAGE_HEIGHT = 60; //验证码图片的高
    private static final String[] CAPTCHA_FONT_NAMES = {"Georgia"}; // 字体数组
    // 验证码数组
    private static final String CAPTCHA_CHAR_CODES = "23456789abcdefghjkmnpqrstuvwxyzABCDEFGHJKMNPQRSTUVWXYZ";

    private String text;                //用来保存验证码的文本内容
    private final Random random = new Random();    //获取随机数对象

    /**
     * 获取随机的颜色
     *
     * @return Color
     */
    private Color randomColor() {
        int r = this.random.nextInt(225);  // 这里为什么是225，因为当r，g，b都为255时，即为白色，为了好辨认，需要颜色深一点。
        int g = this.random.nextInt(225);
        int b = this.random.nextInt(225);
        return new Color(r, g, b);            // 返回一个随机颜色
    }

    /**
     * 获取随机字体
     *
     * @return Font
     */
    private Font randomFont() {
        String fontName = CAPTCHA_FONT_NAMES[0];
        int size = random.nextInt(10) + 24;    //随机获取字体的大小
        return new Font(fontName, Font.ITALIC, size);   //返回一个随机的字体
    }

    /**
     * 获取随机字符
     *
     * @return Char
     */
    private char randomChar() {
        int index = random.nextInt(CAPTCHA_CHAR_CODES.length());
        return CAPTCHA_CHAR_CODES.charAt(index);
    }

    /**
     * 画干扰线，验证码干扰线用来防止计算机解析图片
     *
     * @param image 图片
     */
    private void drawLine(BufferedImage image) {
        int num = random.nextInt(10); //定义干扰线的数量
        Graphics2D g = (Graphics2D) image.getGraphics();
        for (int i = 0; i < num; i++) {
            int x1 = random.nextInt(CAPTCHA_IMAGE_WIDTH);
            int y1 = random.nextInt(CAPTCHA_IMAGE_HEIGHT);
            int x2 = random.nextInt(CAPTCHA_IMAGE_WIDTH);
            int y2 = random.nextInt(CAPTCHA_IMAGE_HEIGHT);
            g.setColor(randomColor());
            g.drawLine(x1, y1, x2, y2);
        }
    }

    /**
     * 创建图片的方法
     *
     * @return BufferedImage
     */
    private BufferedImage createImage() {
        // 创建图片缓冲区
        BufferedImage image = new BufferedImage(CAPTCHA_IMAGE_WIDTH, CAPTCHA_IMAGE_HEIGHT, BufferedImage.TYPE_INT_RGB);
        // 获取画笔
        Graphics2D g = (Graphics2D) image.getGraphics();
        // 设置背景色随机
        g.setColor(new Color(255, 255, random.nextInt(245) + 10));
        g.fillRect(0, 0, CAPTCHA_IMAGE_WIDTH, CAPTCHA_IMAGE_HEIGHT);
        // 返回一个图片
        return image;
    }

    /**
     * 获取验证码图片的方法
     *
     * @return BufferedImage
     */
    public BufferedImage getImage() {
        BufferedImage image = createImage();
        Graphics2D g = (Graphics2D) image.getGraphics(); //获取画笔
        StringBuilder sb = new StringBuilder();
        for (int i = 0; i < 4; i++) {
            String s = randomChar() + "";      // 随机生成字符，因为只有画字符串的方法，没有画字符的方法，所以需要将字符变成字符串再画
            sb.append(s);                      // 添加到StringBuilder里面
            float x = i * 1.0F * CAPTCHA_IMAGE_WIDTH / 4;   // 定义字符的x坐标
            g.setFont(randomFont());           // 设置字体，随机
            g.setColor(randomColor());         // 设置颜色，随机
            g.drawString(s, x, CAPTCHA_IMAGE_HEIGHT - 5.0f);
        }
        this.text = sb.toString();
        drawLine(image);
        return image;
    }

    /**
     * 获取验证码文本的方法
     *
     * @return
     */
    public String getText() {
        return text;
    }
}
